# Licensed to Modin Development Team under one or more contributor license agreements.
# See the NOTICE file distributed with this work for additional information regarding
# copyright ownership.  The Modin Development Team licenses this file to you under the
# Apache License, Version 2.0 (the "License"); you may not use this file except in
# compliance with the License.  You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software distributed under
# the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
# ANY KIND, either express or implied. See the License for the specific language
# governing permissions and limitations under the License.

import polars
import polars.testing

import modin.polars as pl


def test_init_roundtrip():
    data = {"a": [1, 2, 3], "b": [4, 5, 6]}
    df = pl.DataFrame(data)
    polars_df = polars.DataFrame(data)
    to_polars = polars.from_pandas(df._query_compiler.to_pandas())
    polars.testing.assert_frame_equal(polars_df, to_polars)
